Core Apps » ChatSecure:Android

//

//  OTRMessagesViewController.m

//  Off the Record

//

//  Created by David Chiles on 5/12/14.

//  Copyright (c) 2014 Chris Ballinger. All rights reserved.

//

#import "OTRMessagesViewController.h"

#import "OTRDatabaseView.h"

#import "OTRDatabaseManager.h"

#import "OTRLog.h"

#import "OTRBuddy.h"

#import "OTRAccount.h"

#import "OTRMessage+JSQMessageData.h"

@import JSQMessagesViewController;

@import MobileCoreServices;

#import "OTRProtocolManager.h"

#import "OTRXMPPTorAccount.h"

#import "OTRXMPPManager.h"

#import "OTRLockButton.h"

#import "OTRButtonView.h"

@import OTRAssets;

#import "OTRTitleSubtitleView.h"

@import OTRKit;

@import FormatterKit;

#import "OTRImages.h"

#import "UIActivityViewController+ChatSecure.h"

#import "OTRUtilities.h"

#import "OTRProtocolManager.h"

#import "OTRColors.h"

#import "JSQMessagesCollectionViewCell+ChatSecure.h"

@import BButton;

#import "OTRAttachmentPicker.h"

#import "OTRImageItem.h"

#import "OTRVideoItem.h"

#import "OTRAudioItem.h"

@import JTSImageViewController;

#import "OTRAudioControlsView.h"

#import "OTRPlayPauseProgressView.h"

#import "OTRAudioPlaybackController.h"

#import "OTRMediaFileManager.h"

#import "OTRMediaServer.h"

#import "UIImage+ChatSecure.h"

#import "OTRBaseLoginViewController.h"

#import <ChatSecureCore/ChatSecureCore-Swift.h>

#import "OTRYapMessageSendAction.h"

#import "UIViewController+ChatSecure.h"

#import "OTRBuddyCache.h"

#import "OTRTextItem.h"

#import "OTRHTMLItem.h"

#import "OTRFileItem.h"

@import YapDatabase;

@import PureLayout;

@import KVOController;

@import AVFoundation;

@import MediaPlayer;

static NSTimeInterval const kOTRMessageSentDateShowTimeInterval = 5 * 60;

static NSUInteger const kOTRMessagePageSize = 50;

typedef NS_ENUM(int, OTRDropDownType) {

    OTRDropDownTypeNone          = 0,

    OTRDropDownTypeEncryption    = 1,

    OTRDropDownTypePush          = 2

};

@interface OTRMessagesViewController () <UITextViewDelegate, OTRAttachmentPickerDelegate, OTRYapViewHandlerDelegateProtocol, OTRMessagesCollectionViewFlowLayoutSizeProtocol, OTRRoomOccupantsViewControllerDelegate> {

    JSQMessagesAvatarImage *_warningAvatarImage;

    JSQMessagesAvatarImage *_accountAvatarImage;

    JSQMessagesAvatarImage *_buddyAvatarImage;

}

@property (nonatomic, strong) OTRYapViewHandler *viewHandler;

@property (nonatomic, strong) JSQMessagesBubbleImage *outgoingBubbleImage;

@property (nonatomic, strong) JSQMessagesBubbleImage *incomingBubbleImage;

@property (nonatomic, weak) id didFinishGeneratingPrivateKeyNotificationObject;

@property (nonatomic, weak) id messageStateDidChangeNotificationObject;

@property (nonatomic, weak) id pendingApprovalDidChangeNotificationObject;

@property (nonatomic, weak) id deviceListUpdateNotificationObject;

@property (nonatomic ,strong) UIBarButtonItem *lockBarButtonItem;

@property (nonatomic, strong) OTRLockButton *lockButton;

@property (nonatomic, strong) OTRButtonView *buttonDropdownView;

@property (nonatomic, strong) OTRAttachmentPicker *attachmentPicker;

@property (nonatomic, strong) OTRAudioPlaybackController *audioPlaybackController;

@property (nonatomic, strong) NSTimer *lastSeenRefreshTimer;

@property (nonatomic, strong) UIView *jidForwardingHeaderView;

@property (nonatomic) BOOL loadingMessages;

@property (nonatomic) BOOL messageRangeExtended;

@property (nonatomic, strong) NSIndexPath *currentIndexPath;

@property (nonatomic, strong) id currentMessage;

@property (nonatomic, strong) NSCache *messageSizeCache;

@end

@implementation OTRMessagesViewController

- (instancetype)initWithNibName:(NSString *)nibNameOrNil bundle:(NSBundle *)nibBundleOrNil

{

    if (self = [super initWithNibName:nibNameOrNil bundle:nibBundleOrNil]) {

        self.senderId = @"";

        self.senderDisplayName = @"";

        _state = [[MessagesViewControllerState alloc] init];

        self.messageSizeCache = [NSCache new];

        self.messageSizeCache.countLimit = kOTRMessagePageSize;

        self.messageRangeExtended = NO;

    }

    return self;

}

#pragma - mark Lifecylce Methods

- (void) dealloc {

    [self.lastSeenRefreshTimer invalidate];

    [[NSNotificationCenter defaultCenter] removeObserver:self];

}

- (void)viewDidLoad

{

    [super viewDidLoad];

    self.automaticallyScrollsToMostRecentMessage = YES;

     ////// bubbles //////

    JSQMessagesBubbleImageFactory *bubbleImageFactory = [[JSQMessagesBubbleImageFactory alloc] init];

    self.outgoingBubbleImage = [bubbleImageFactory outgoingMessagesBubbleImageWithColor:[UIColor jsq_messageBubbleBlueColor]];

    self.incomingBubbleImage = [bubbleImageFactory incomingMessagesBubbleImageWithColor:[UIColor jsq_messageBubbleLightGrayColor]];

    ////// TitleView //////

    OTRTitleSubtitleView *titleView = [self titleView];

    [self refreshTitleView:titleView];

    self.navigationItem.titleView = titleView;

    ////// Send Button //////

    self.sendButton = [JSQMessagesToolbarButtonFactory defaultSendButtonItem];

    ////// Attachment Button //////

    self.inputToolbar.contentView.leftBarButtonItem = nil;

    self.cameraButton = [UIButton buttonWithType:UIButtonTypeRoundedRect];

    self.cameraButton.titleLabel.font = [UIFont fontWithName:kFontAwesomeFont size:20];

    self.cameraButton.titleLabel.textAlignment = NSTextAlignmentCenter;

    [self.cameraButton setTitle:[NSString fa_stringForFontAwesomeIcon:FACamera] forState:UIControlStateNormal];

    self.cameraButton.frame = CGRectMake(0, 0, 32, 32);

    ////// Microphone Button //////

    self.microphoneButton = [UIButton buttonWithType:UIButtonTypeRoundedRect];

    self.microphoneButton.frame = CGRectMake(0, 0, 32, 32);

    self.microphoneButton.titleLabel.font = [UIFont fontWithName:kFontAwesomeFont size:20];

    self.microphoneButton.titleLabel.textAlignment = NSTextAlignmentCenter;

    [self.microphoneButton setTitle:[NSString fa_stringForFontAwesomeIcon:FAMicrophone]

          forState:UIControlStateNormal];

    self.audioPlaybackController = [[OTRAudioPlaybackController alloc] init];

    ////// TextViewUpdates //////

    [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(receivedTextViewChangedNotification:) name:UITextViewTextDidChangeNotification object:self.inputToolbar.contentView.textView];

    /** Setup databse view handler*/

    self.viewHandler = [[OTRYapViewHandler alloc] initWithDatabaseConnection:[OTRDatabaseManager sharedInstance].longLivedReadOnlyConnection databaseChangeNotificationName:[DatabaseNotificationName LongLivedTransactionChanges]];

    self.viewHandler.delegate = self;

    ///Custom Layout to account for no bubble cells

    OTRMessagesCollectionViewFlowLayout *layout = [[OTRMessagesCollectionViewFlowLayout alloc] init];

    layout.sizeDelegate = self;

    self.collectionView.collectionViewLayout = layout;

    ///"Loading Earlier" header view

    [self.collectionView registerNib:[UINib nibWithNibName:@"OTRMessagesLoadingView" bundle:OTRAssets.resourcesBundle]

          forSupplementaryViewOfKind:UICollectionElementKindSectionHeader

                 withReuseIdentifier:[JSQMessagesLoadEarlierHeaderView headerReuseIdentifier]];

    //Subscribe to changes in encryption state

    __weak typeof(self)weakSelf = self;

    [self.KVOController observe:self.state keyPath:NSStringFromSelector(@selector(messageSecurity)) options:NSKeyValueObservingOptionInitial|NSKeyValueObservingOptionNew block:^(id  _Nullable observer, id  _Nonnull object, NSDictionary<NSString *,id> * _Nonnull change) {

        __typeof__(self) strongSelf = weakSelf;

        if (!strongSelf) { return; }

        if ([object isKindOfClass:[MessagesViewControllerState class]]) {

            MessagesViewControllerState *state = (MessagesViewControllerState*)object;

            NSString * placeHolderString = nil;

            switch (state.messageSecurity) {

                case OTRMessageTransportSecurityPlaintext:

                case OTRMessageTransportSecurityPlaintextWithOTR:

                    placeHolderString = SEND_PLAINTEXT_STRING();

                    break;

                case OTRMessageTransportSecurityOTR:

                    placeHolderString = [NSString stringWithFormat:SEND_ENCRYPTED_STRING(),@"OTR"];

                    break;

                case OTRMessageTransportSecurityOMEMO:

                    placeHolderString = [NSString stringWithFormat:SEND_ENCRYPTED_STRING(),@"OMEMO"];;

                    break;

                default:

                    placeHolderString = [NSBundle jsq_localizedStringForKey:@"new_message"];

                    break;

            }

            strongSelf.inputToolbar.contentView.textView.placeHolder = placeHolderString;

            [self didUpdateState];

        }

    }];

}

- (void)viewDidAppear:(BOOL)animated

{

    [super viewDidAppear:animated];

    [self tryToMarkAllMessagesAsRead];

    // This is a hack to attempt fixing https://github.com/ChatSecure/ChatSecure-iOS/issues/657

    dispatch_after(dispatch_time(DISPATCH_TIME_NOW, (int64_t)(0.25 * NSEC_PER_SEC)), dispatch_get_main_queue(), ^{

        [self scrollToBottomAnimated:animated];

    });

    self.loadingMessages = NO;

}

- (void)viewWillAppear:(BOOL)animated

{

    self.currentIndexPath = nil;

    [super viewWillAppear:animated];

    [[UIApplication sharedApplication] setStatusBarHidden:NO];

    if (self.lastSeenRefreshTimer) {

        [self.lastSeenRefreshTimer invalidate];

        _lastSeenRefreshTimer = nil;

    }

    _lastSeenRefreshTimer = [NSTimer scheduledTimerWithTimeInterval:5 target:self selector:@selector(refreshTitleTimerUpdate:) userInfo:nil repeats:YES];

    __weak typeof(self)weakSelf = self;

    void (^refreshGeneratingLock)(OTRAccount *) = ^void(OTRAccount * account) {

        __strong typeof(weakSelf)strongSelf = weakSelf;

        __block NSString *accountKey = nil;

        [strongSelf.readOnlyDatabaseConnection readWithBlock:^(YapDatabaseReadTransaction * _Nonnull transaction) {

            accountKey = [strongSelf buddyWithTransaction:transaction].accountUniqueId;

        }];

        if ([account.uniqueId isEqualToString:accountKey]) {

            [strongSelf updateEncryptionState];

        }

    };

    self.didFinishGeneratingPrivateKeyNotificationObject = [[NSNotificationCenter defaultCenter] addObserverForName:OTRDidFinishGeneratingPrivateKeyNotification object:nil queue:[NSOperationQueue mainQueue] usingBlock:^(NSNotification *note) {

        if ([note.object isKindOfClass:[OTRAccount class]]) {

            refreshGeneratingLock(note.object);

        }

    }];

    self.messageStateDidChangeNotificationObject = [[NSNotificationCenter defaultCenter] addObserverForName:OTRMessageStateDidChangeNotification object:nil queue:[NSOperationQueue mainQueue] usingBlock:^(NSNotification *note) {

        __strong typeof(weakSelf)strongSelf = weakSelf;

        if ([note.object isKindOfClass:[OTRBuddy class]]) {

            OTRBuddy *notificationBuddy = note.object;

            __block NSString *buddyKey = nil;

            [strongSelf.readOnlyDatabaseConnection readWithBlock:^(YapDatabaseReadTransaction * _Nonnull transaction) {

                buddyKey = [strongSelf buddyWithTransaction:transaction].uniqueId;

            }];

            if ([notificationBuddy.uniqueId isEqualToString:buddyKey]) {

                [strongSelf updateEncryptionState];

            }

        }

    }];

    if ([self.threadKey length]) {

        [self.viewHandler.keyCollectionObserver observe:self.threadKey collection:self.threadCollection];

        [self updateViewWithKey:self.threadKey collection:self.threadCollection];

        [self.viewHandler setup:OTRFilteredChatDatabaseViewExtensionName groups:@[self.threadKey]];

        if(![self.inputToolbar.contentView.textView.text length]) {

            [self moveLastComposingTextForThreadKey:self.threadKey colleciton:self.threadCollection toTextView:self.inputToolbar.contentView.textView];

        }

    }

    self.loadingMessages = YES;

    [self.collectionView reloadData];

}

- (void)viewWillDisappear:(BOOL)animated

{

    [super viewWillDisappear:animated];

    [self.lastSeenRefreshTimer invalidate];

    self.lastSeenRefreshTimer = nil;

    [self saveCurrentMessageText:self.inputToolbar.contentView.textView.text threadKey:self.threadKey colleciton:self.threadCollection];

    [[NSNotificationCenter defaultCenter] removeObserver:self.messageStateDidChangeNotificationObject];

    [[NSNotificationCenter defaultCenter] removeObserver:self.didFinishGeneratingPrivateKeyNotificationObject];

    // [self.inputToolbar.contentView.textView resignFirstResponder];

}

- (void)viewDidDisappear:(BOOL)animated

{

    [super viewDidDisappear:animated];

    _warningAvatarImage = nil;

    _accountAvatarImage = nil;

    _buddyAvatarImage = nil;

}

- (void)viewWillTransitionToSize:(CGSize)size withTransitionCoordinator:(id<UIViewControllerTransitionCoordinator>)coordinator {

    [super viewWillTransitionToSize:size withTransitionCoordinator:coordinator];

    // After the transition is done, we need to reset the size caches and relayout

    // Do this using the technique in https://stackoverflow.com/questions/26943808/ios-how-to-run-a-function-after-device-has-rotated-swift

    [coordinator animateAlongsideTransition:nil completion:^(id<UIViewControllerTransitionCoordinatorContext>  _Nonnull context) {

        [self.messageSizeCache removeAllObjects];

        [self.collectionView.collectionViewLayout invalidateLayoutWithContext:[JSQMessagesCollectionViewFlowLayoutInvalidationContext context]];

    }];

}

#pragma - mark Setters & getters

- (OTRAttachmentPicker *)attachmentPicker

{

    if (!_attachmentPicker) {

        _attachmentPicker = [[OTRAttachmentPicker alloc] initWithParentViewController:self delegate:self];

    }

    return _attachmentPicker;

}

- (NSArray*) indexPathsToCount:(NSUInteger)count {

    NSMutableArray *indexPaths = [NSMutableArray arrayWithCapacity:count];

    for (NSUInteger i = 0; i < count; i++) {

        NSIndexPath *indexPath = [NSIndexPath indexPathForRow:i inSection:0];

        [indexPaths addObject:indexPath];

    }

    return indexPaths;

}

- (nullable id<OTRThreadOwner>)threadObjectWithTransaction:(nonnull YapDatabaseReadTransaction *)transaction {

    if (!self.threadKey || !self.threadCollection || !transaction) { return nil; }

    id object = [transaction objectForKey:self.threadKey inCollection:self.threadCollection];

    if ([object conformsToProtocol:@protocol(OTRThreadOwner)]) {

        return object;

    }

    return nil;

}

- (nullable OTRBuddy *)buddyWithTransaction:(nonnull YapDatabaseReadTransaction *)transaction {

    id <OTRThreadOwner> object = [self threadObjectWithTransaction:transaction];

    if ([object isKindOfClass:[OTRBuddy class]]) {

        return (OTRBuddy *)object;

    }

    return nil;

}

- (nullable OTRXMPPRoom *)roomWithTransaction:(nonnull YapDatabaseReadTransaction *)transaction {

    id <OTRThreadOwner> object = [self threadObjectWithTransaction:transaction];

    if ([object isKindOfClass:[OTRXMPPRoom class]]) {

        return (OTRXMPPRoom *)object;

    }

    return nil;

}

- (nullable OTRAccount *)accountWithTransaction:(nonnull YapDatabaseReadTransaction *)transaction {

    id <OTRThreadOwner> thread =  [self threadObjectWithTransaction:transaction];

    if (!thread) { return nil; }

    OTRAccount *account = [OTRAccount fetchObjectWithUniqueID:[thread threadAccountIdentifier] transaction:transaction];

    return account;

}

- (void)setThreadKey:(NSString *)key collection:(NSString *)collection

{

    self.currentIndexPath = nil;

    NSString *oldKey = self.threadKey;

    NSString *oldCollection = self.threadCollection;

    self.threadKey = key;

    self.threadCollection = collection;

    [self.readOnlyDatabaseConnection readWithBlock:^(YapDatabaseReadTransaction * _Nonnull transaction) {

        self.senderId = [[self threadObjectWithTransaction:transaction] threadAccountIdentifier];

    }];

    // Clear out old state (don't just alloc a new object, we have KVOs attached to this!)

    self.state.canSendMedia = NO;

    self.state.canKnock = NO;

    self.state.messageSecurity = OTRMessageTransportSecurityInvalid;

    self.state.hasText = NO;

    self.state.isThreadOnline = NO;

    self.showTypingIndicator = NO;

    // This is set to nil so the refreshTitleView: method knows to reset username instead of last seen time

    [self titleView].subtitleLabel.text = nil;

    if (![oldKey isEqualToString:key] || ![oldCollection isEqualToString:collection]) {

        [self saveCurrentMessageText:self.inputToolbar.contentView.textView.text threadKey:oldKey colleciton:oldCollection];

        self.inputToolbar.contentView.textView.text = nil;

        [self receivedTextViewChanged:self.inputToolbar.contentView.textView];

    }

    [self.viewHandler.keyCollectionObserver stopObserving:oldKey collection:oldCollection];

    if (self.threadKey && self.threadCollection) {

        [self.viewHandler.keyCollectionObserver observe:self.threadKey collection:self.threadCollection];

        [self updateViewWithKey:self.threadKey collection:self.threadCollection];

        [self.viewHandler setup:OTRFilteredChatDatabaseViewExtensionName groups:@[self.threadKey]];

        [self moveLastComposingTextForThreadKey:self.threadKey colleciton:self.threadCollection toTextView:self.inputToolbar.contentView.textView];

    } else {

        // Reset the view handler

        self.viewHandler = [[OTRYapViewHandler alloc] initWithDatabaseConnection:[OTRDatabaseManager sharedInstance].longLivedReadOnlyConnection databaseChangeNotificationName:[DatabaseNotificationName LongLivedTransactionChanges]];

        self.viewHandler.delegate = self;

        self.senderDisplayName = @"";

        self.senderId = @"";

    }

    [self.collectionView reloadData];

    // Profile Info Button

    [self setupInfoButton];

    [self updateEncryptionState];

    [self updateJIDForwardingHeader];

    __weak typeof(self)weakSelf = self;

    if (self.pendingApprovalDidChangeNotificationObject == nil) {

        self.pendingApprovalDidChangeNotificationObject = [[NSNotificationCenter defaultCenter] addObserverForName:OTRBuddyPendingApprovalDidChangeNotification object:nil queue:[NSOperationQueue mainQueue] usingBlock:^(NSNotification *note) {

            __strong typeof(weakSelf)strongSelf = weakSelf;

            OTRXMPPBuddy *notificationBuddy = [note.userInfo objectForKey:@"buddy"];

            __block NSString *buddyKey = nil;

            [strongSelf.readOnlyDatabaseConnection readWithBlock:^(YapDatabaseReadTransaction * _Nonnull transaction) {

                buddyKey = [strongSelf buddyWithTransaction:transaction].uniqueId;

            }];

            if ([notificationBuddy.uniqueId isEqualToString:buddyKey]) {

                [strongSelf fetchOMEMODeviceList];

                [strongSelf sendPresenceProbe];

            }

        }];

    }

    if (self.deviceListUpdateNotificationObject == nil) {

        self.deviceListUpdateNotificationObject = [[NSNotificationCenter defaultCenter] addObserverForName:OTROMEMOSignalCoordinator.DeviceListUpdateNotificationName object:nil queue:[NSOperationQueue mainQueue] usingBlock:^(NSNotification * _Nonnull note) {

            __strong typeof(weakSelf)strongSelf = weakSelf;

            XMPPJID *notificationJid = [note.userInfo objectForKey:@"jid"];

            __block NSString *buddyUser = nil;

            [strongSelf.readOnlyDatabaseConnection readWithBlock:^(YapDatabaseReadTransaction * _Nonnull transaction) {

                buddyUser = [strongSelf buddyWithTransaction:transaction].username;

            }];

            if (notificationJid != nil && [notificationJid.bare isEqualToString:buddyUser]) {

                [strongSelf updateEncryptionState];

            }

        }];

    }

    [self sendPresenceProbe];

    [self fetchOMEMODeviceList];

}

- (YapDatabaseConnection *)readOnlyDatabaseConnection

{

    if (!_readOnlyDatabaseConnection) {

        _readOnlyDatabaseConnection = [OTRDatabaseManager sharedInstance].readOnlyDatabaseConnection;

    }

    return _readOnlyDatabaseConnection;

}

- (YapDatabaseConnection *)readWriteDatabaseConnection

{

    if (!_readWriteDatabaseConnection) {

            _readWriteDatabaseConnection = [OTRDatabaseManager sharedInstance].readWriteDatabaseConnection;

    }

    return _readWriteDatabaseConnection;

}

- (nullable OTRXMPPManager *)xmppManagerWithTransaction:(nonnull YapDatabaseReadTransaction *)transaction {

    OTRAccount *account = [self accountWithTransaction:transaction];

    if (!account) { return nil; }

    return (OTRXMPPManager *)[[OTRProtocolManager sharedInstance] protocolForAccount:account];

}

/** Will send a probe to fetch last seen */

- (void) sendPresenceProbe {

    __block OTRXMPPManager *xmpp = nil;

    __block OTRXMPPBuddy *buddy = nil;

    [self.readOnlyDatabaseConnection readWithBlock:^(YapDatabaseReadTransaction * _Nonnull transaction) {

        xmpp = [self xmppManagerWithTransaction:transaction];

        buddy = (OTRXMPPBuddy*)[self buddyWithTransaction:transaction];

    }];

    if (!xmpp || ![buddy isKindOfClass:[OTRXMPPBuddy class]] || buddy.pendingApproval) { return; }

    [xmpp sendPresenceProbeForBuddy:buddy];

}

- (void)updateViewWithKey:(NSString *)key collection:(NSString *)collection

{

    if ([collection isEqualToString:[OTRBuddy collection]]) {

        __block OTRBuddy *buddy = nil;

        __block OTRAccount *account = nil;

        [self.readOnlyDatabaseConnection readWithBlock:^(YapDatabaseReadTransaction * _Nonnull transaction) {

            buddy = [OTRBuddy fetchObjectWithUniqueID:key transaction:transaction];

            account = [OTRAccount fetchObjectWithUniqueID:buddy.accountUniqueId transaction:transaction];

        }];

        //Update UI now

        if (buddy.chatState == OTRChatStateComposing || buddy.chatState == OTRChatStatePaused) {

            self.showTypingIndicator = YES;

        }

        else {

            self.showTypingIndicator = NO;

        }

        // Update Buddy Status

        BOOL previousState = self.state.isThreadOnline;

        self.state.isThreadOnline = buddy.status != OTRThreadStatusOffline;

        [self didUpdateState];

        //Update Buddy knock status

        //Async because this calls down to the database and iterates over a relation. Might slowdown the UI if on main thread

        __weak __typeof__(self) weakSelf = self;

        dispatch_async(dispatch_get_global_queue(DISPATCH_QUEUE_PRIORITY_DEFAULT, 0), ^{

            __typeof__(self) strongSelf = weakSelf;

            __block BOOL canKnock = [[[OTRProtocolManager sharedInstance].pushController pushStorage] numberOfTokensForBuddy:buddy.uniqueId createdByThisAccount:NO] > 0;

            dispatch_async(dispatch_get_main_queue(), ^{

                if (canKnock != strongSelf.state.canKnock) {

                    strongSelf.state.canKnock = canKnock;

                    [strongSelf didUpdateState];

                }

            });

        });

        [self refreshTitleView:[self titleView]];

        // Auto-inititate OTR when contact comes online

        if (!previousState && self.state.isThreadOnline) {

            [[OTRProtocolManager sharedInstance].encryptionManager maybeRefreshOTRSessionForBuddyKey:key collection:collection];

        }

    } else if ([collection isEqualToString:[OTRXMPPRoom collection]]) {

        __block OTRXMPPRoom *room = nil;

        [self.readOnlyDatabaseConnection readWithBlock:^(YapDatabaseReadTransaction * _Nonnull transaction) {

            room = [OTRXMPPRoom fetchObjectWithUniqueID:key transaction:transaction];

        }];

        self.state.isThreadOnline = room.currentStatus != OTRThreadStatusOffline;

        [self didUpdateState];

        [self refreshTitleView:[self titleView]];

    }

    [self tryToMarkAllMessagesAsRead];

}

- (void)tryToMarkAllMessagesAsRead {

    // Set all messages as read

    if ([self otr_isVisible]) {

        __weak __typeof__(self) weakSelf = self;

        __block id <OTRThreadOwner>threadOwner = nil;

        __block NSArray <id <OTRMessageProtocol>>* unreadMessages = nil;

        [self.readOnlyDatabaseConnection asyncReadWithBlock:^(YapDatabaseReadTransaction * _Nonnull transaction) {

            threadOwner = [weakSelf threadObjectWithTransaction:transaction];

            if (!threadOwner) { return; }

            unreadMessages = [transaction allUnreadMessagesForThread:threadOwner];

        } completionBlock:^{

            if ([unreadMessages count] == 0) {

                return;

            }

            //Mark as read

            NSMutableArray <id <OTRMessageProtocol>>*toBeSaved = [[NSMutableArray alloc] init];

            [unreadMessages enumerateObjectsUsingBlock:^(id<OTRMessageProtocol>  _Nonnull obj, NSUInteger idx, BOOL * _Nonnull stop) {

                if ([obj isKindOfClass:[OTRIncomingMessage class]]) {

                    OTRIncomingMessage *message = [((OTRIncomingMessage *)obj) copy];

                    message.read = YES;

                    [toBeSaved addObject:message];

                } else if ([obj isKindOfClass:[OTRXMPPRoomMessage class]]) {

                    OTRXMPPRoomMessage *message = [((OTRXMPPRoomMessage *)obj) copy];

                    message.read = YES;

                    [toBeSaved addObject:message];

                }

            }];

            [weakSelf.readWriteDatabaseConnection asyncReadWriteWithBlock:^(YapDatabaseReadWriteTransaction * _Nonnull transaction) {

                [toBeSaved enumerateObjectsUsingBlock:^(id<OTRMessageProtocol>  _Nonnull obj, NSUInteger idx, BOOL * _Nonnull stop) {

                    [transaction setObject:obj forKey:[obj messageKey] inCollection:[obj messageCollection]];

                }];

                [transaction touchObjectForKey:[threadOwner threadIdentifier] inCollection:[threadOwner threadCollection]];

            }];

        }];

    }

}

- (OTRTitleSubtitleView * __nonnull)titleView {

    UIView *titleView = self.navigationItem.titleView;

    if ([titleView isKindOfClass:[OTRTitleSubtitleView class]]) {

        return  (OTRTitleSubtitleView*)titleView;

    }

    return [[OTRTitleSubtitleView alloc] initWithFrame:CGRectMake(0, 0, 200, 44)];

}

- (void)refreshTitleTimerUpdate:(NSTimer*)timer {

    [self refreshTitleView:[self titleView]];

}

/** Updates the title view with the current thread information on this view controller*/

- (void)refreshTitleView:(OTRTitleSubtitleView *)titleView

{

    __block id<OTRThreadOwner> thread = nil;

    __block OTRAccount *account = nil;

    [self.readOnlyDatabaseConnection readWithBlock:^(YapDatabaseReadTransaction * _Nonnull transaction) {

        thread = [self threadObjectWithTransaction:transaction];

        account =  [self accountWithTransaction:transaction];

    }];

    titleView.titleLabel.text = [thread threadName];

    UIImage *statusImage = nil;

    if ([thread isKindOfClass:[OTRBuddy class]]) {

        OTRBuddy *buddy = (OTRBuddy*)thread;

        UIColor *color = [buddy avatarBorderColor];

        if (color) { // only show online status

            statusImage = [OTRImages circleWithRadius:50

                                      lineWidth:0

                                      lineColor:nil

                                      fillColor:color];

        }

        dispatch_block_t refreshTimeBlock = ^{

            __block OTRBuddy *buddy = nil;

            [self.readOnlyDatabaseConnection readWithBlock:^(YapDatabaseReadTransaction * _Nonnull transaction) {

                buddy = (OTRBuddy*)[self threadObjectWithTransaction:transaction];

            }];

            if (![buddy isKindOfClass:[OTRBuddy class]]) {

                return;

            }

            NSDate *lastSeen = [OTRBuddyCache.shared lastSeenDateForBuddy:buddy];

            OTRThreadStatus status = [OTRBuddyCache.shared threadStatusForBuddy:buddy];

            if (!lastSeen) {

                titleView.subtitleLabel.text = buddy.username;

                return;

            }

            TTTTimeIntervalFormatter *tf = [[TTTTimeIntervalFormatter alloc] init];

            tf.presentTimeIntervalMargin = 60;

            tf.usesAbbreviatedCalendarUnits = YES;

            NSTimeInterval lastSeenInterval = [lastSeen timeIntervalSinceDate:[NSDate date]];

            NSString *labelString = nil;

            if (status == OTRThreadStatusAvailable) {

                labelString = buddy.username;

            } else {

                labelString = [NSString stringWithFormat:@"%@ %@", ACTIVE_STRING(), [tf stringForTimeInterval:lastSeenInterval]];

            }

            titleView.subtitleLabel.text = labelString;

        };

        // Set the username if nothing else is set.

        // This should be cleared out when buddy is changed

        if (!titleView.subtitleLabel.text) {

            titleView.subtitleLabel.text = buddy.username;

        }

        // Show an "Last seen 11 min ago" in title bar after brief delay

        dispatch_after(dispatch_time(DISPATCH_TIME_NOW, (int64_t)(3 * NSEC_PER_SEC)), dispatch_get_main_queue(), ^{

            refreshTimeBlock();

        });

    } else if ([thread isGroupThread]) {

        titleView.subtitleLabel.text = GROUP_CHAT_STRING();

    } else {

        titleView.subtitleLabel.text = nil;

    }

    titleView.titleImageView.image = statusImage;

}

/**

 This generates a UIAlertAction where the handler fetches the outgoing message (optionaly duplicates). Then if media message resend media message. If not update messageSecurityInfo and date and create new sending action.

 */

- (UIAlertAction *)resendOutgoingMessageActionForMessageKey:(NSString *)messageKey

                                          messageCollection:(NSString *)messageCollection

                                readWriteDatabaseConnection:(YapDatabaseConnection*)databaseConnection

                                                      title:(NSString *)title

{

    UIAlertAction *action = [UIAlertAction actionWithTitle:title style:UIAlertActionStyleDefault handler:^(UIAlertAction * _Nonnull action) {

        [databaseConnection asyncReadWriteWithBlock:^(YapDatabaseReadWriteTransaction * _Nonnull transaction) {

            id object = [[transaction objectForKey:messageKey inCollection:messageCollection] copy];

            id<OTRMessageProtocol> message = nil;

            if ([object conformsToProtocol:@protocol(OTRMessageProtocol)]) {

                message = (id<OTRMessageProtocol>)object;

            } else {

                return;

            }

            // Messages that never sent properly don't need to be duplicated client-side

            NSError *messageError = message.messageError;

            message = [message duplicateMessage];

            message.messageError = nil;

            message.messageSecurity = self.state.messageSecurity;

            message.messageDate = [NSDate date];

            [message saveWithTransaction:transaction];

            // We only need to re-upload failed media messages

            // otherwise just resend the URL directly

            if (message.messageMediaItemKey.length &&

                (!message.messageText.length || messageError)) {

                OTRMediaItem *mediaItem = [OTRMediaItem fetchObjectWithUniqueID:message.messageMediaItemKey transaction:transaction];

                [self sendMediaItem:mediaItem data:nil message:message transaction:transaction];

            } else {

                OTRYapMessageSendAction *sendingAction = [OTRYapMessageSendAction sendActionForMessage:message date:message.messageDate];

                [sendingAction saveWithTransaction:transaction];

            }

        }];

    }];

    return action;

}

- (nonnull UIAlertAction *)viewProfileAction {

    return [UIAlertAction actionWithTitle:VIEW_PROFILE_STRING() style:UIAlertActionStyleDefault handler:^(UIAlertAction * _Nonnull action) {

        [self infoButtonPressed:action];

    }];

}

- (nonnull UIAlertAction *)cancleAction {

    return [UIAlertAction actionWithTitle:CANCEL_STRING()

                                    style:UIAlertActionStyleCancel

                                  handler:nil];

}

- (NSArray <UIAlertAction *>*)actionForMessage:(id<OTRMessageProtocol>)message {

    NSMutableArray <UIAlertAction *>*actions = [[NSMutableArray alloc] init];

    if (!message.isMessageIncoming) {

        // This is an outgoing message so we can offer to resend

        UIAlertAction *resendAction = [self resendOutgoingMessageActionForMessageKey:message.messageKey messageCollection:message.messageCollection readWriteDatabaseConnection:self.readWriteDatabaseConnection  title:RESEND_STRING()];

        [actions addObject:resendAction];

    }

    if (![message isKindOfClass:[OTRXMPPRoomMessage class]]) {

        [actions addObject:[self viewProfileAction]];

    }

    NSArray<UIAlertAction*> *mediaActions = [UIAlertAction actionsForMediaMessage:message sourceView:self.view viewController:self];

    [actions addObjectsFromArray:mediaActions];

    [actions addObject:[self cancleAction]];

    return actions;

}

- (void)didTapAvatar:(id<OTRMessageProtocol>)message sender:(id)sender {

    NSError *error =  [message messageError];

    NSString *title = nil;

    NSString *alertMessage = nil;

    NSString * sendingType = UNENCRYPTED_STRING();

    switch (self.state.messageSecurity) {

        case OTRMessageTransportSecurityOTR:

            sendingType = @"OTR";

            break;

        case OTRMessageTransportSecurityOMEMO:

            sendingType = @"OMEMO";

            break;

        default:

            break;

    }

    if ([message isKindOfClass:[OTROutgoingMessage class]]) {

        title = RESEND_MESSAGE_TITLE();

        alertMessage = [NSString stringWithFormat:RESEND_DESCRIPTION_STRING(),sendingType];

    }

    if (error) {

        NSUInteger otrFingerprintError = 32872;

        title = ERROR_STRING();

        alertMessage = error.localizedDescription;

        if (error.code == otrFingerprintError) {

            alertMessage = NO_DEVICES_BUDDY_ERROR_STRING();

        }

        if([message isKindOfClass:[OTROutgoingMessage class]]) {

            //If it's an outgoing message the error title should be that we were unable to send the message.

            title = UNABLE_TO_SEND_STRING();

            NSString *resendDescription = [NSString stringWithFormat:RESEND_DESCRIPTION_STRING(),sendingType];

            alertMessage = [alertMessage stringByAppendingString:[NSString stringWithFormat:@"\n%@",resendDescription]];

            //If this is an error about not having a trusted identity then we should offer to connect to the

            if (error.code == OTROMEMOErrorNoDevicesForBuddy ||

                error.code == OTROMEMOErrorNoDevices ||

                error.code == otrFingerprintError) {

                alertMessage = [alertMessage stringByAppendingString:[NSString stringWithFormat:@"\n%@",VIEW_PROFILE_DESCRIPTION_STRING()]];

            }

        }

    }

    if (![self isMessageTrusted:message]) {

        title = UNTRUSTED_DEVICE_STRING();

        if ([message isMessageIncoming]) {

            alertMessage = UNTRUSTED_DEVICE_REVEIVED_STRING();

        } else {

            alertMessage = UNTRUSTED_DEVICE_SENT_STRING();

        }

        alertMessage = [alertMessage stringByAppendingString:[NSString stringWithFormat:@"\n%@",VIEW_PROFILE_DESCRIPTION_STRING()]];

    }

    NSArray <UIAlertAction*>*actions = [self actionForMessage:message];

    if ([actions count] > 0) {

        UIAlertController *alertController = [UIAlertController alertControllerWithTitle:title message:alertMessage preferredStyle:UIAlertControllerStyleActionSheet];

        [actions enumerateObjectsUsingBlock:^(UIAlertAction * _Nonnull obj, NSUInteger idx, BOOL * _Nonnull stop) {

            [alertController addAction:obj];

        }];

        if ([sender isKindOfClass:[UIView class]]) {

            UIView *sourceView = sender;

            alertController.popoverPresentationController.sourceView = sourceView;

            alertController.popoverPresentationController.sourceRect = sourceView.bounds;

        }

        [self presentViewController:alertController animated:YES completion:nil];

    }

}

- (BOOL)isMessageTrusted:(id <OTRMessageProtocol>)message {

    BOOL trusted = YES;

    if (![message isKindOfClass:[OTRBaseMessage class]]) {

        return trusted;

    }

    OTRBaseMessage *baseMessage = (OTRBaseMessage *)message;

    if (baseMessage.messageSecurityInfo.messageSecurity == OTRMessageTransportSecurityOTR) {

        NSData *otrFingerprintData = baseMessage.messageSecurityInfo.otrFingerprint;

        if ([otrFingerprintData length]) {

            trusted = [[[OTRProtocolManager sharedInstance].encryptionManager otrFingerprintForKey:self.threadKey collection:self.threadCollection fingerprint:otrFingerprintData] isTrusted];

        }

    } else if (baseMessage.messageSecurityInfo.messageSecurity == OTRMessageTransportSecurityOMEMO) {

        NSString *omemoDeviceYapKey = baseMessage.messageSecurityInfo.omemoDeviceYapKey;

        NSString *omemoDeviceYapCollection = baseMessage.messageSecurityInfo.omemoDeviceYapCollection;

        __block OTROMEMODevice *device = nil;

        [self.readOnlyDatabaseConnection readWithBlock:^(YapDatabaseReadTransaction * _Nonnull transaction) {

            device = [transaction objectForKey:omemoDeviceYapKey inCollection:omemoDeviceYapCollection];

        }];

        if(device != nil) {

            trusted = [device isTrusted];

        }

    }

    return trusted;

}

- (BOOL) isGroupChat {

    __block OTRXMPPRoom *room = nil;

    [self.readOnlyDatabaseConnection readWithBlock:^(YapDatabaseReadTransaction * _Nonnull transaction) {

        room = [self roomWithTransaction:transaction];

    }];

    return (room != nil);

}

#pragma - mark Profile Button Methods

- (void)setupInfoButton {

    if ([self isGroupChat]) {

        UIBarButtonItem *barButtonItem = [[UIBarButtonItem alloc] initWithImage:[UIImage imageNamed:@"112-group" inBundle:[OTRAssets resourcesBundle] compatibleWithTraitCollection:nil] style:UIBarButtonItemStylePlain target:self action:@selector(didSelectOccupantsButton:)];

        self.navigationItem.rightBarButtonItem = barButtonItem;

    } else {

        UIButton* infoButton = [UIButton buttonWithType:UIButtonTypeInfoLight];

        infoButton.accessibilityIdentifier = @"profileButton";

        [infoButton addTarget:self action:@selector(infoButtonPressed:) forControlEvents:UIControlEventTouchUpInside];

        self.navigationItem.rightBarButtonItem = [[UIBarButtonItem alloc] initWithCustomView:infoButton];

    }

}

- (void) infoButtonPressed:(id)sender {

    __block OTRAccount *account = nil;

    __block OTRBuddy *buddy = nil;

    [self.readOnlyDatabaseConnection readWithBlock:^(YapDatabaseReadTransaction * _Nonnull transaction) {

        account = [self accountWithTransaction:transaction];

        buddy = [self buddyWithTransaction:transaction];

    }];

    if (!account || !buddy) {

        return;

    }

    // Hack to manually re-fetch OMEMO devicelist because PEP sucks

    // TODO: Ideally this should be moved to some sort of manual refresh in the Profile view

    [self fetchOMEMODeviceList];

    XLFormDescriptor *form = [UserProfileViewController profileFormDescriptorForAccount:account buddies:@[buddy] connection:self.readOnlyDatabaseConnection];

    UserProfileViewController *verify = [[UserProfileViewController alloc] initWithAccountKey:account.uniqueId connection:self.readOnlyDatabaseConnection form:form];

    verify.completionBlock = ^{

        [self updateEncryptionState];

    };

    UINavigationController *verifyNav = [[UINavigationController alloc] initWithRootViewController:verify];

    verifyNav.modalPresentationStyle = UIModalPresentationFormSheet;

    [self presentViewController:verifyNav animated:YES completion:nil];

}

- (void)didSelectOccupantsButton:(id)sender {

    UIStoryboard *storyboard = [UIStoryboard storyboardWithName:@"OTRRoomOccupants" bundle:[OTRAssets resourcesBundle]];

    OTRRoomOccupantsViewController *occupantsVC = [storyboard instantiateViewControllerWithIdentifier:@"roomOccupants"];

    occupantsVC.delegate = self;

    [occupantsVC setupViewHandlerWithDatabaseConnection:[OTRDatabaseManager sharedInstance].longLivedReadOnlyConnection roomKey:self.threadKey];

    [self.navigationController pushViewController:occupantsVC animated:YES];

}

// Hack to manually re-fetch OMEMO devicelist because PEP sucks

// TODO: Ideally this should be moved to some sort of manual refresh in the Profile view

-(void) fetchOMEMODeviceList {

    __block OTRAccount *account = nil;

    __block OTRBuddy *buddy = nil;

    [self.readOnlyDatabaseConnection readWithBlock:^(YapDatabaseReadTransaction * _Nonnull transaction) {

        account = [self accountWithTransaction:transaction];

        buddy = [self buddyWithTransaction:transaction];

    }];

    if (!account || !buddy) {

        return;

    }

    id manager = [[OTRProtocolManager sharedInstance] protocolForAccount:account];

    if ([manager isKindOfClass:[OTRXMPPManager class]]) {

        XMPPJID *jid = [XMPPJID jidWithString:buddy.username];

        OTRXMPPManager *xmpp = manager;

        [xmpp.omemoSignalCoordinator.omemoModule fetchDeviceIdsForJID:jid elementId:nil];

    }

}

- (UIBarButtonItem *)rightBarButtonItem

{

    if (!self.lockBarButtonItem) {

        self.lockBarButtonItem = [[UIBarButtonItem alloc] initWithCustomView:self.lockButton];

    }

    return self.lockBarButtonItem;

}

-(void)updateEncryptionState

{

    if ([self isGroupChat]) {

        __block OTRXMPPManager *xmpp = nil;

        [self.readOnlyDatabaseConnection asyncReadWithBlock:^(YapDatabaseReadTransaction * _Nonnull transaction) {

            xmpp = [self xmppManagerWithTransaction:transaction];

        } completionBlock:^{

            BOOL canSendMedia = NO;

            // Check for XEP-0363 HTTP upload

            // TODO: move this check elsewhere so it isnt dependent on refreshing crypto state

            if (xmpp != nil && xmpp.fileTransferManager.canUploadFiles) {

                canSendMedia = YES;

            }

            self.state.canSendMedia = canSendMedia;

            self.state.messageSecurity = OTRMessageTransportSecurityPlaintext;

            [self didUpdateState];

        }];

    } else {

        __block OTRBuddy *buddy = nil;

        __block OTRAccount *account = nil;

        __block OTRXMPPManager *xmpp = nil;

        __block OTRMessageTransportSecurity messageSecurity = OTRMessageTransportSecurityInvalid;

        [self.readOnlyDatabaseConnection asyncReadWithBlock:^(YapDatabaseReadTransaction * _Nonnull transaction) {

            buddy = [self buddyWithTransaction:transaction];

            account = [buddy accountWithTransaction:transaction];

            xmpp = [self xmppManagerWithTransaction:transaction];

            messageSecurity = [buddy preferredTransportSecurityWithTransaction:transaction];

        } completionBlock:^{

            BOOL canSendMedia = NO;

            // Check for XEP-0363 HTTP upload

            // TODO: move this check elsewhere so it isnt dependent on refreshing crypto state

            if (xmpp != nil && xmpp.fileTransferManager.canUploadFiles) {

                canSendMedia = YES;

            }

            if (!buddy || !account || !xmpp || (messageSecurity == OTRMessageTransportSecurityInvalid)) {

                DDLogError(@"updateEncryptionState error: missing parameters");

            } else {

                OTRKitMessageState messageState = [[OTRProtocolManager sharedInstance].encryptionManager.otrKit messageStateForUsername:buddy.username accountName:account.username protocol:account.protocolTypeString];

                if (messageState == OTRKitMessageStateEncrypted &&

                    buddy.status != OTRThreadStatusOffline) {

                    // If other side supports OTR, assume OTRDATA is possible

                    canSendMedia = YES;

                }

            }

            self.state.canSendMedia = canSendMedia;

            self.state.messageSecurity = messageSecurity;

            [self didUpdateState];

        }];

    }

}

- (void)setupAccessoryButtonsWithMessageState:(OTRKitMessageState)messageState buddyStatus:(OTRThreadStatus)status textViewHasText:(BOOL)hasText

{

    self.inputToolbar.contentView.rightBarButtonItem = self.sendButton;

    self.inputToolbar.sendButtonLocation = JSQMessagesInputSendButtonLocationRight;

    self.inputToolbar.contentView.leftBarButtonItem = nil;

}

- (void)connectButtonPressed:(id)sender

{

    [self hideDropdownAnimated:YES completion:nil];

    __block OTRAccount *account = nil;

    [self.readOnlyDatabaseConnection readWithBlock:^(YapDatabaseReadTransaction * _Nonnull transaction) {

        account = [self accountWithTransaction:transaction];

    }];

    if (account == nil) {

        return;

    }

    //If we have the password then we can login with that password otherwise show login UI to enter password

    if ([account.password length]) {

        [[OTRProtocolManager sharedInstance] loginAccount:account userInitiated:YES];

    } else {

        OTRBaseLoginViewController *loginViewController = [[OTRBaseLoginViewController alloc] initWithAccount:account];

        UINavigationController *nav = [[UINavigationController alloc] initWithRootViewController:loginViewController];

        nav.modalPresentationStyle = UIModalPresentationFormSheet;

        [self presentViewController:nav animated:YES completion:nil];

    }

}

#pragma - mark  dropDown Methods

- (void)showDropdownWithTitle:(NSString *)title buttons:(NSArray *)buttons animated:(BOOL)animated tag:(NSInteger)tag

{

    NSTimeInterval duration = 0.3;

    if (!animated) {

        duration = 0.0;

    }

    self.buttonDropdownView = [[OTRButtonView alloc] initWithTitle:title buttons:buttons];

    self.buttonDropdownView.tag = tag;

    CGFloat height = [OTRButtonView heightForTitle:title width:self.view.bounds.size.width buttons:buttons];

    [self.view addSubview:self.buttonDropdownView];

    [self.buttonDropdownView autoSetDimension:ALDimensionHeight toSize:height];

    [self.buttonDropdownView autoPinEdgeToSuperviewEdge:ALEdgeLeading];

    [self.buttonDropdownView autoPinEdgeToSuperviewEdge:ALEdgeTrailing];

    self.buttonDropdownView.topLayoutConstraint = [self.buttonDropdownView autoPinToTopLayoutGuideOfViewController:self withInset:height*-1];

    [self.buttonDropdownView layoutIfNeeded];

    [UIView animateWithDuration:duration animations:^{

        self.buttonDropdownView.topLayoutConstraint.constant = 0.0;

        [self.buttonDropdownView layoutIfNeeded];

    } completion:nil];

}

- (void)hideDropdownAnimated:(BOOL)animated completion:(void (^)(void))completion

{

    if (!self.buttonDropdownView) {

        if (completion) {

            completion();

        }

    }

    else {

        NSTimeInterval duration = 0.3;

        if (!animated) {

            duration = 0.0;

        }

        [UIView animateWithDuration:duration animations:^{

            CGFloat height = self.buttonDropdownView.frame.size.height;

            self.buttonDropdownView.topLayoutConstraint.constant = height*-1;

            [self.buttonDropdownView layoutIfNeeded];

        } completion:^(BOOL finished) {

            if (finished) {

                [self.buttonDropdownView removeFromSuperview];

                self.buttonDropdownView = nil;

            }

            if (completion) {

                completion();

            }

        }];

    }

}

- (void)saveCurrentMessageText:(NSString *)text threadKey:(NSString *)key colleciton:(NSString *)collection

{

    if (![key length] || ![collection length]) {

        return;

    }

    [self.readWriteDatabaseConnection asyncReadWriteWithBlock:^(YapDatabaseReadWriteTransaction * _Nonnull transaction) {

        id <OTRThreadOwner> thread = [[transaction objectForKey:key inCollection:collection] copy];

        if (thread == nil) {

            // this can happen when we've just approved a contact, then the thread key

            // might have changed.

            return;

        }

        [thread setCurrentMessageText:text];

        [transaction setObject:thread forKey:key inCollection:collection];

        //Send inactive chat State

        OTRAccount *account = [OTRAccount fetchObjectWithUniqueID:[thread threadAccountIdentifier] transaction:transaction];

        OTRXMPPManager *xmppManager = (OTRXMPPManager *)[[OTRProtocolManager sharedInstance] protocolForAccount:account];

        if (![text length]) {

            [xmppManager sendChatState:OTRChatStateInactive withBuddyID:[thread threadIdentifier]];

        }

    }];

}

//* Takes the current value out of the thread object and sets it to the text view and nils out result*/

- (void)moveLastComposingTextForThreadKey:(NSString *)key colleciton:(NSString *)collection toTextView:(UITextView *)textView {

    if (![key length] || ![collection length] || !textView) {

        return;

    }

    __block id <OTRThreadOwner> thread = nil;

    [self.readOnlyDatabaseConnection readWithBlock:^(YapDatabaseReadTransaction * _Nonnull transaction) {

        thread = [[transaction objectForKey:key inCollection:collection] copy];

    }];

    // Don't remove text you're already composing

    NSString *oldThreadText = [thread currentMessageText];

    if (!textView.text.length && oldThreadText.length) {

        textView.text = oldThreadText;

        [self receivedTextViewChanged:textView];

    }

    if (oldThreadText.length) {

        [thread setCurrentMessageText:nil];

        [self.readWriteDatabaseConnection asyncReadWriteWithBlock:^(YapDatabaseReadWriteTransaction * _Nonnull transaction) {

            [transaction setObject:thread forKey:key inCollection:collection];

        }];

    }

}

- (id <OTRMessageProtocol,JSQMessageData>)messageAtIndexPath:(NSIndexPath *)indexPath

{

    // Multiple invocations with the same indexPath tend to come in groups, no need to hit the DB each time.

    // Even though the object is cached, the row ID calculation still takes time

    if (![indexPath isEqual:self.currentIndexPath]) {

        self.currentIndexPath = indexPath;

        self.currentMessage = [self.viewHandler object:indexPath];

    }

    return self.currentMessage;

}

/**

 * Updates the flexible range of the DB connection.

 * @param reset When NO, adds kOTRMessagePageSize to the range length, when YES resets the length to the kOTRMessagePageSize

 */

- (void)updateRangeOptions:(BOOL)reset

{

    YapDatabaseViewRangeOptions *options = [self.viewHandler.mappings rangeOptionsForGroup:self.threadKey];

    if (reset) {

        if (options != nil && !self.messageRangeExtended) {

            return;

        }

        options = [YapDatabaseViewRangeOptions flexibleRangeWithLength:kOTRMessagePageSize

                                                                offset:0

                                                                  from:YapDatabaseViewEnd];

        self.messageSizeCache.countLimit = kOTRMessagePageSize;

        self.messageRangeExtended = NO;

    } else {

        options = [options copyWithNewLength:options.length + kOTRMessagePageSize];

        self.messageSizeCache.countLimit += kOTRMessagePageSize;

        self.messageRangeExtended = YES;

    }

    [self.viewHandler.mappings setRangeOptions:options forGroup:self.threadKey];

    self.loadingMessages = YES;

    CGFloat distanceToBottom = self.collectionView.contentSize.height - self.collectionView.contentOffset.y;

    [self.collectionView reloadData];

    [self.readOnlyDatabaseConnection readWithBlock:^(YapDatabaseReadTransaction *_Nonnull transaction) {

        NSUInteger shownCount = [self.viewHandler.mappings numberOfItemsInGroup:self.threadKey];

        NSUInteger totalCount = [[transaction ext:OTRFilteredChatDatabaseViewExtensionName] numberOfItemsInGroup:self.threadKey];

        [self setShowLoadEarlierMessagesHeader:shownCount < totalCount];

    }];

    if (!reset) {

        [self.collectionView.collectionViewLayout invalidateLayout];

        [self.collectionView layoutSubviews];

        self.collectionView.contentOffset = CGPointMake(0, self.collectionView.contentSize.height - distanceToBottom);

    }

    self.loadingMessages = NO;

}

- (BOOL)showDateAtIndexPath:(NSIndexPath *)indexPath

{

    BOOL showDate = NO;

    if (indexPath.row == 0) {

        showDate = YES;

    }

    else {

        id <OTRMessageProtocol> currentMessage = [self messageAtIndexPath:indexPath];

        id <OTRMessageProtocol> previousMessage = [self messageAtIndexPath:[NSIndexPath indexPathForItem:indexPath.row-1 inSection:indexPath.section]];

        NSTimeInterval timeDifference = [[currentMessage messageDate] timeIntervalSinceDate:[previousMessage messageDate]];

        if (timeDifference > kOTRMessageSentDateShowTimeInterval) {

            showDate = YES;

        }

    }

    return showDate;

}

- (BOOL)showSenderDisplayNameAtIndexPath:(NSIndexPath *)indexPath {

    id<OTRMessageProtocol,JSQMessageData> message = [self messageAtIndexPath:indexPath];

    if(![self.threadCollection isEqualToString:[OTRXMPPRoom collection]]) {

        return NO;

    }

    if ([[message senderId] isEqualToString:self.senderId]) {

        return NO;

    }

    if(indexPath.row -1 >= 0) {

        NSIndexPath *previousIndexPath = [NSIndexPath indexPathForRow:indexPath.row - 1 inSection:indexPath.section];

        id<OTRMessageProtocol,JSQMessageData> previousMessage = [self messageAtIndexPath:previousIndexPath];

        if ([[previousMessage senderId] isEqualToString:message.senderId]) {

            return NO;

        }

    }

    return YES;

}

- (BOOL)isPushMessageAtIndexPath:(NSIndexPath *)indexPath {

    id message = [self messageAtIndexPath:indexPath];

    return [message isKindOfClass:[PushMessage class]];

}

- (void)receivedTextViewChangedNotification:(NSNotification *)notification

{

    //Check if the text state changes from having some text to some or vice versa

    UITextView *textView = notification.object;

    [self receivedTextViewChanged:textView];

}

- (void)receivedTextViewChanged:(UITextView *)textView {

    BOOL hasText = [textView.text length] > 0;

    if(hasText != self.state.hasText) {

        self.state.hasText = hasText;

        [self didUpdateState];

    }

    //Everytime the textview has text and a notification comes through we are 'typing' otherwise we are done typing

    if (hasText) {

        [self isTyping];

    } else {

        [self didFinishTyping];

    }

    return;

}

#pragma - mark Update UI

- (void)didUpdateState {

}

- (void)isTyping {

}

- (void)didFinishTyping {

}

#pragma - mark Sending Media Items

- (void)sendMediaItem:(OTRMediaItem *)mediaItem data:(NSData *)data message:(id<OTRMessageProtocol>)message transaction:(YapDatabaseReadWriteTransaction *)transaction

{

    id<OTRThreadOwner> thread = [self threadObjectWithTransaction:transaction];

    OTRXMPPManager *xmpp = [self xmppManagerWithTransaction:transaction];

    if (!message || !thread || !xmpp) {

        DDLogError(@"Error sending file due to bad paramters");

        return;

    }

    if (data) {

        thread.lastMessageIdentifier = message.messageKey;

        [thread saveWithTransaction:transaction];

    }

    // XEP-0363

    [xmpp.fileTransferManager sendWithMediaItem:mediaItem prefetchedData:data message:message];

    [mediaItem touchParentMessageWithTransaction:transaction];

}

#pragma - mark Media Display Methods

- (void)showImage:(OTRImageItem *)imageItem fromCollectionView:(JSQMessagesCollectionView *)collectionView atIndexPath:(NSIndexPath *)indexPath

{

    //FIXME: Possible for image to not be in cache?

    UIImage *image = [OTRImages imageWithIdentifier:imageItem.uniqueId];

    JTSImageInfo *imageInfo = [[JTSImageInfo alloc] init];

    imageInfo.image = image;

    UICollectionViewCell *cell = [collectionView cellForItemAtIndexPath:indexPath];

    if ([cell isKindOfClass:[JSQMessagesCollectionViewCell class]]) {

        UIView *cellContainterView = ((JSQMessagesCollectionViewCell *)cell).messageBubbleContainerView;

        imageInfo.referenceRect = cellContainterView.bounds;

        imageInfo.referenceView = cellContainterView;

        imageInfo.referenceCornerRadius = 10;

    }

    JTSImageViewController *imageViewer = [[JTSImageViewController alloc]

                                           initWithImageInfo:imageInfo

                                           mode:JTSImageViewControllerMode_Image

                                           backgroundStyle:JTSImageViewControllerBackgroundOption_Blurred];

    [imageViewer showFromViewController:self transition:JTSImageViewControllerTransition_FromOriginalPosition];

}

- (void)showVideo:(OTRVideoItem *)videoItem fromCollectionView:(JSQMessagesCollectionView *)collectionView atIndexPath:(NSIndexPath *)indexPath

{

    if (videoItem.filename) {

        NSURL *videoURL = [[OTRMediaServer sharedInstance] urlForMediaItem:videoItem buddyUniqueId:self.threadKey];

        MPMoviePlayerViewController *moviePlayerViewController = [[MPMoviePlayerViewController alloc] initWithContentURL:videoURL];

        [self presentViewController:moviePlayerViewController animated:YES completion:nil];

    }

}

- (void)playOrPauseAudio:(OTRAudioItem *)audioItem fromCollectionView:(JSQMessagesCollectionView *)collectionView atIndexPath:(NSIndexPath *)indexPath

{

    NSError *error = nil;

    if  ([audioItem.uniqueId isEqualToString:self.audioPlaybackController.currentAudioItem.uniqueId]) {

        if  ([self.audioPlaybackController isPlaying]) {

            [self.audioPlaybackController pauseCurrentlyPlaying];

        }

        else {

            [self.audioPlaybackController resumeCurrentlyPlaying];

        }

    }

    else {

        [self.audioPlaybackController stopCurrentlyPlaying];

        OTRAudioControlsView *audioControls = [self audioControllsfromCollectionView:collectionView atIndexPath:indexPath];

        [self.audioPlaybackController attachAudioControlsView:audioControls];

        [self.audioPlaybackController playAudioItem:audioItem buddyUniqueId:self.threadKey error:&error];

    }

    if (error) {

         DDLogError(@"Audio Playback Error: %@",error);

    }

}

- (OTRAudioControlsView *)audioControllsfromCollectionView:(JSQMessagesCollectionView *)collectionView atIndexPath:(NSIndexPath *)indexPath {

    UICollectionViewCell *cell = [collectionView cellForItemAtIndexPath:indexPath];

    if ([cell isKindOfClass:[JSQMessagesCollectionViewCell class]]) {

        UIView *mediaView = ((JSQMessagesCollectionViewCell *)cell).mediaView;

        UIView *view = [mediaView viewWithTag:kOTRAudioControlsViewTag];

        if ([view isKindOfClass:[OTRAudioControlsView class]]) {

            return (OTRAudioControlsView *)view;

        }

    }

    return nil;

}

#pragma MARK - OTRMessagesCollectionViewFlowLayoutSizeProtocol methods

- (BOOL)hasBubbleSizeForCellAtIndexPath:(NSIndexPath *)indexPath {

    return ![self isPushMessageAtIndexPath:indexPath];

}

#pragma mark - JSQMessagesViewController method overrides

- (UICollectionViewCell *)collectionView:(JSQMessagesCollectionView *)collectionView cellForItemAtIndexPath:(NSIndexPath *)indexPath

{

    JSQMessagesCollectionViewCell *cell = (JSQMessagesCollectionViewCell *)[super collectionView:collectionView cellForItemAtIndexPath:indexPath];

    //Fixes times when there needs to be two lines (date & knock sent) and doesn't seem to affect one line instances

    cell.cellTopLabel.numberOfLines = 0;

    id <OTRMessageProtocol>message = [self messageAtIndexPath:indexPath];

    __block OTRXMPPAccount *account = nil;

    [self.readOnlyDatabaseConnection readWithBlock:^(YapDatabaseReadTransaction * _Nonnull transaction) {

        account = (OTRXMPPAccount*)[self accountWithTransaction:transaction];

    }];

    UIColor *textColor = nil;

    if ([message isMessageIncoming]) {

        textColor = [UIColor blackColor];

    }

    else {

        textColor = [UIColor whiteColor];

    }

    if (cell.textView != nil)

        cell.textView.textColor = textColor;

	// Do not allow clickable links for Tor accounts to prevent information leakage

    // Could be better to move this information to the message object to not need to do a database read.

    if ([account isKindOfClass:[OTRXMPPTorAccount class]]) {

        cell.textView.dataDetectorTypes = UIDataDetectorTypeNone;

    }

    else {

        cell.textView.dataDetectorTypes = UIDataDetectorTypeLink;

        cell.textView.linkTextAttributes = @{ NSForegroundColorAttributeName : textColor,

                                              NSUnderlineStyleAttributeName : @(NSUnderlineStyleSingle | NSUnderlinePatternSolid) };

    }

    if ([[message messageMediaItemKey] isEqualToString:self.audioPlaybackController.currentAudioItem.uniqueId]) {

        UIView *view = [cell.mediaView viewWithTag:kOTRAudioControlsViewTag];

        if ([view isKindOfClass:[OTRAudioControlsView class]]) {

            [self.audioPlaybackController attachAudioControlsView:(OTRAudioControlsView *)view];

        }

    }

    // Needed for link interaction

    cell.textView.delegate = self;

    return cell;

}

- (BOOL)collectionView:(UICollectionView *)collectionView canPerformAction:(SEL)action forItemAtIndexPath:(NSIndexPath *)indexPath withSender:(id)sender

{

    if (action == @selector(delete:)) {

        return YES;

    }

    return [super collectionView:collectionView canPerformAction:action forItemAtIndexPath:indexPath withSender:sender];

}

- (void)didPressSendButton:(UIButton *)button withMessageText:(NSString *)text senderId:(NSString *)senderId senderDisplayName:(NSString *)senderDisplayName date:(NSDate *)date

{

    if(!text.length) {

        return;

    }

    self.navigationController.providesPresentationContextTransitionStyle = YES;

    self.navigationController.definesPresentationContext = YES;

    //0. Clear out message text immediately

    //   This is to prevent the scenario where multiple messages get sent because the message text isn't cleared out

    //   due to aggregated touch events during UI pauses.

    //   A side effect is that sent messages may not appear in the UI immediately

    [self finishSendingMessage];

    __block id<OTRMessageProtocol> message = nil;

    __block OTRXMPPManager *xmpp = nil;

    [self.readOnlyDatabaseConnection readWithBlock:^(YapDatabaseReadTransaction * _Nonnull transaction) {

        id<OTRThreadOwner> thread = [self threadObjectWithTransaction:transaction];

        message = [thread outgoingMessageWithText:text transaction:transaction];

        xmpp = [self xmppManagerWithTransaction:transaction];

    }];

    if (!message || !xmpp) { return; }

    [xmpp enqueueMessage:message];

}

- (void)didPressAccessoryButton:(UIButton *)sender

{

    if ([sender isEqual:self.cameraButton]) {

        [self.attachmentPicker showAlertControllerFromSourceView:sender withCompletion:nil];

    }

}

- (void)collectionView:(UICollectionView *)collectionView performAction:(SEL)action forItemAtIndexPath:(NSIndexPath *)indexPath withSender:(id)sender

{

    if (action == @selector(delete:)) {

        [self deleteMessageAtIndexPath:indexPath];

    }

    else {

        [super collectionView:collectionView performAction:action forItemAtIndexPath:indexPath withSender:sender];

    }

}

- (CGSize)collectionView:(UICollectionView *)collectionView layout:(UICollectionViewLayout *)collectionViewLayout sizeForItemAtIndexPath:(NSIndexPath *)indexPath

{

    id <OTRMessageProtocol, JSQMessageData> message = [self messageAtIndexPath:indexPath];

    NSNumber *key = @(message.messageHash);

    NSValue *sizeValue = [self.messageSizeCache objectForKey:key];

    if (sizeValue != nil) {

        return [sizeValue CGSizeValue];

    }

    // Although JSQMessagesBubblesSizeCalculator has its own cache, its size is fixed and quite small, so it quickly chokes on scrolling into the past

    CGSize size = [super collectionView:collectionView layout:collectionViewLayout sizeForItemAtIndexPath:indexPath];

    // The height of the first cell might change: on loading additional messages the date label most likely will disappear

    if (indexPath.row > 0) {

        [self.messageSizeCache setObject:[NSValue valueWithCGSize:size] forKey:key];

    }

    return size;

}

#pragma - mark UIPopoverPresentationControllerDelegate Methods

- (void)prepareForPopoverPresentation:(UIPopoverPresentationController *)popoverPresentationController {

    // Without setting this, there will be a crash on iPad

    // This delegate is set in the OTRAttachmentPicker

    popoverPresentationController.sourceView = self.cameraButton;

}

- (void)sendPhoto:(UIImage *)photo asJPEG:(BOOL)asJPEG shouldResize:(BOOL)shouldResize {

    NSParameterAssert(photo);

    if (!photo) { return; }

    __block OTRXMPPManager *xmpp = nil;

    __block id<OTRThreadOwner> thread = nil;

    [self.readOnlyDatabaseConnection readWithBlock:^(YapDatabaseReadTransaction * _Nonnull transaction) {

        xmpp = [self xmppManagerWithTransaction:transaction];

        thread = [self threadObjectWithTransaction:transaction];

    }];

    NSParameterAssert(xmpp);

    NSParameterAssert(thread);

    if (!xmpp || !thread) { return; }

    [xmpp.fileTransferManager sendWithImage:photo thread:thread];

}

#pragma - mark OTRAttachmentPickerDelegate Methods

- (void)attachmentPicker:(OTRAttachmentPicker *)attachmentPicker gotPhoto:(UIImage *)photo withInfo:(NSDictionary *)info

{

    [self sendPhoto:photo asJPEG:YES shouldResize:YES];

}

- (void)attachmentPicker:(OTRAttachmentPicker *)attachmentPicker gotVideoURL:(NSURL *)videoURL

{

    if (!videoURL) { return; }

    __block OTRXMPPManager *xmpp = nil;

    __block id<OTRThreadOwner> thread = nil;

    [self.readOnlyDatabaseConnection readWithBlock:^(YapDatabaseReadTransaction * _Nonnull transaction) {

        xmpp = [self xmppManagerWithTransaction:transaction];

        thread = [self threadObjectWithTransaction:transaction];

    }];

    NSParameterAssert(xmpp);

    NSParameterAssert(thread);

    if (!xmpp || !thread) { return; }

    [xmpp.fileTransferManager sendWithVideoURL:videoURL thread:thread];

}

- (NSArray <NSString *>*)attachmentPicker:(OTRAttachmentPicker *)attachmentPicker preferredMediaTypesForSource:(UIImagePickerControllerSourceType)source

{

    return @[(NSString*)kUTTypeImage];

}

- (void)sendAudioFileURL:(NSURL *)url

{

    if (!url) { return; }

    __block OTRXMPPManager *xmpp = nil;

    __block id<OTRThreadOwner> thread = nil;

    [self.readOnlyDatabaseConnection readWithBlock:^(YapDatabaseReadTransaction * _Nonnull transaction) {

        xmpp = [self xmppManagerWithTransaction:transaction];

        thread = [self threadObjectWithTransaction:transaction];

    }];

    NSParameterAssert(xmpp);

    NSParameterAssert(thread);

    if (!xmpp || !thread) { return; }

    [xmpp.fileTransferManager sendWithAudioURL:url thread:thread];

}

- (void)sendImageFilePath:(NSString *)filePath asJPEG:(BOOL)asJPEG shouldResize:(BOOL)shouldResize

{

    [self sendPhoto:[UIImage imageWithContentsOfFile:filePath] asJPEG:asJPEG shouldResize:shouldResize];

}

#pragma - mark UIScrollViewDelegate Methods

- (void)scrollViewWillBeginDragging:(UIScrollView *)scrollView

{

    [self hideDropdownAnimated:YES completion:nil];

}